--- title: Title keywords: fastai sidebar: home_sidebar nb_path: "02d05a Testing Branch Clusters Circle Swiss roll.ipynb" ---
{% raw %}
{% endraw %} {% raw %} 
from directed_graphs.datasets import directed_sinh_branch, directed_swiss_roll_uniform, directed_circle, static_clusters
from directed_graphs.datasets import plot_directed_2d, plot_directed_3d
from directed_graphs.datasets import affinity_grid_search
{% endraw %}

Branch

{% raw %} 
X, flow, labels = directed_sinh_branch(num_nodes=300, xscale=2, yscale=1, sigma=0.5, xtilt=np.pi/4, ytilt=np.pi/4)
plot_directed_3d(X, flow, labels)
# sigma = 1, flow_strength = 2
{% endraw %} {% raw %} 
sigmas = [0.5, 1, 3, 5]
flow_strengths = [1, 2, 5, 10]
affinity_grid_search(X, flow, sigmas, flow_strengths)
{% endraw %}

Swiss Roll

{% raw %} 
X, flow, labels = directed_swiss_roll_uniform(num_nodes=1000, num_spirals=2.5, radius=1, height=50, xtilt=0, ytilt=0)
plot_directed_3d(X, flow, labels)
# sigma = 1, flow_strength = 1
{% endraw %} {% raw %} 
sigmas = [0.5, 1, 3, 5]
flow_strengths = [1, 2, 5, 10]
affinity_grid_search(X, flow, sigmas, flow_strengths)
{% endraw %}

2d Circle

{% raw %} 
X, flow, labels = directed_circle(num_nodes=200, radius=1, xtilt=0, ytilt=0, twodim=True)
plot_directed_2d(X, flow, labels)
# sigma = 0.5, flow_strength = 0.5
{% endraw %} {% raw %} 
sigmas = [0.25, 0.5, 1, 3]
flow_strengths = [0.5, 1, 2, 3]
affinity_grid_search(X, flow, sigmas, flow_strengths)
{% endraw %}

3d Circle

{% raw %} 
X, flow, labels = directed_circle(num_nodes=200, radius=1, xtilt=np.pi/4, ytilt=0)
plot_directed_3d(X, flow, labels)
# sigma = 0.5, flow_strength = 0.5
{% endraw %} {% raw %} 
sigmas = [0.25, 0.5, 1, 3]
flow_strengths = [0.5, 1, 2, 3]
affinity_grid_search(X, flow, sigmas, flow_strengths)
{% endraw %}

Static Clutsers

{% raw %} 
X, flow, lables = static_clusters(num_nodes=800, num_clusters=5, radius=1, sigma=0.2, xtilt=np.pi/4, ytilt=np.pi/4)
plot_directed_3d(X, flow, lables)
# sigma = 0.1, flow_strength = 1
{% endraw %} {% raw %} 
sigmas = [0.1, 0.2, 0.3, 1]
flow_strengths = [0.5, 1, 2, 3]
affinity_grid_search(X, flow, sigmas, flow_strengths)
{% endraw %}